import csv
import sqlite3

connection = sqlite3.connect('../db.sqlite3')
cursor = connection.cursor()

create_table = '''
create table data_HDD
(
    Name text not null,
    ID text not null,
    PID text not null,
    Platform text not null,
    Price text,
    Capacity text,
    Slot text,
    CacheSize text,
    Platter text,
    RPM text,
    ReadWriteSpeed text,
    ReadWriteSpeed4K text,
    constraint hdd_pk
        primary key (Name,ID,PID,Platform)
);


'''

cursor.execute(create_table)
file = open('../GetInfo/Details/HDDInfo.csv', encoding='utf-8')
contents = csv.reader(file)
insert_records = "INSERT INTO data_HDD (Name,ID,PID,Platform,Price,Capacity,Slot,CacheSize,Platter,RPM,ReadWriteSpeed,ReadWriteSpeed4K) " \
                 "VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"

cursor.executemany(insert_records, contents)
select_all = "SELECT * FROM data_HDD"
rows = cursor.execute(select_all).fetchall()
for r in rows:
    print(r)

# Committing the changes
connection.commit()
connection.close()
